//SPDX-FileCopyrightText: Copyright 2022-2024 深圳市同心圆网络有限公司
//SPDX-License-Identifier: GPL-3.0-only

package admin_idea_api_serv

import (
	"context"

	"atomgit.com/openlinksaas-org/api-server/admin_api_serv/admin_common"
	"atomgit.com/openlinksaas-org/api-server/dao"
	"atomgit.com/openlinksaas-org/api-server/dao/idea_dao"
	"atomgit.com/openlinksaas-org/proto-gen-go.git/idea_store_api"
	"github.com/google/uuid"
	"go.mongodb.org/mongo-driver/mongo"
)

func (apiImpl *IdeaStoreAdminApiImpl) CreateStore(ctx context.Context, req *idea_store_api.AdminCreateStoreRequest) (*idea_store_api.AdminCreateStoreResponse, error) {
	sess, ok := dao.CacheDao.GetAdminSession(ctx, req.AdminSessionId)
	if !ok {
		return &idea_store_api.AdminCreateStoreResponse{
			Code:   idea_store_api.AdminCreateStoreResponse_CODE_WRONG_SESSION,
			ErrMsg: "会话不存在",
		}, nil
	}
	if !sess.AuthOk {
		return &idea_store_api.AdminCreateStoreResponse{
			Code:   idea_store_api.AdminCreateStoreResponse_CODE_NOT_AUTH,
			ErrMsg: "未授权会话",
		}, nil
	}
	adminUserCfg, err := admin_common.GetAdminUserConfig(ctx, apiImpl.adminCfg, sess.UserName)
	if err != nil {
		return nil, err
	}
	if !adminUserCfg.PermInfo.IdeaStorePerm.CreateStore {
		return &idea_store_api.AdminCreateStoreResponse{
			Code:   idea_store_api.AdminCreateStoreResponse_CODE_NO_PERMISSION,
			ErrMsg: "没有权限",
		}, nil
	}

	exist, err := idea_dao.IdeaStoreCateDao.Exist(ctx, req.StoreCateId)
	if err != nil {
		return nil, err
	}
	if !exist {
		return &idea_store_api.AdminCreateStoreResponse{
			Code:   idea_store_api.AdminCreateStoreResponse_CODE_NO_STORE_CATE,
			ErrMsg: "知识点仓库分类不存在",
		}, nil
	}

	ideaStoreId := uuid.NewString()

	dbSess, err := dao.StartSession()
	if err != nil {
		return nil, err
	}
	defer dao.EndSession(dbSess)

	ret, err := dbSess.WithTransaction(ctx, func(sessCtx mongo.SessionContext) (interface{}, error) {
		err = idea_dao.IdeaStoreDao.Insert(sessCtx, &idea_dao.IdeaStoreDbItem{
			IdeaStoreId: ideaStoreId,
			StoreCateId: req.StoreCateId,
			Name:        req.Name,
			Weight:      req.Weight,
			IdeaCount:   0,
		})
		if err != nil {
			return nil, err
		}
		count, err := idea_dao.IdeaStoreDao.CountByCate(sessCtx, req.StoreCateId)
		if err != nil {
			return nil, err
		}
		err = idea_dao.IdeaStoreCateDao.UpdateStoreCount(sessCtx, req.StoreCateId, count)
		if err != nil {
			return nil, err
		}
		return &idea_store_api.AdminCreateStoreResponse{
			Code:        idea_store_api.AdminCreateStoreResponse_CODE_OK,
			IdeaStoreId: ideaStoreId,
		}, nil
	})
	if err != nil {
		return nil, err
	}
	return ret.(*idea_store_api.AdminCreateStoreResponse), nil
}

func (apiImpl *IdeaStoreAdminApiImpl) UpdateStore(ctx context.Context, req *idea_store_api.AdminUpdateStoreRequest) (*idea_store_api.AdminUpdateStoreResponse, error) {
	sess, ok := dao.CacheDao.GetAdminSession(ctx, req.AdminSessionId)
	if !ok {
		return &idea_store_api.AdminUpdateStoreResponse{
			Code:   idea_store_api.AdminUpdateStoreResponse_CODE_WRONG_SESSION,
			ErrMsg: "会话不存在",
		}, nil
	}
	if !sess.AuthOk {
		return &idea_store_api.AdminUpdateStoreResponse{
			Code:   idea_store_api.AdminUpdateStoreResponse_CODE_NOT_AUTH,
			ErrMsg: "未授权会话",
		}, nil
	}
	adminUserCfg, err := admin_common.GetAdminUserConfig(ctx, apiImpl.adminCfg, sess.UserName)
	if err != nil {
		return nil, err
	}
	if !adminUserCfg.PermInfo.IdeaStorePerm.UpdateStore {
		return &idea_store_api.AdminUpdateStoreResponse{
			Code:   idea_store_api.AdminUpdateStoreResponse_CODE_NO_PERMISSION,
			ErrMsg: "没有权限",
		}, nil
	}

	exist, err := idea_dao.IdeaStoreDao.Exist(ctx, req.IdeaStoreId)
	if err != nil {
		return nil, err
	}
	if !exist {
		return &idea_store_api.AdminUpdateStoreResponse{
			Code:   idea_store_api.AdminUpdateStoreResponse_CODE_NO_STORE,
			ErrMsg: "知识点仓库不存在",
		}, nil
	}

	dbSess, err := dao.StartSession()
	if err != nil {
		return nil, err
	}
	defer dao.EndSession(dbSess)

	ret, err := dbSess.WithTransaction(ctx, func(sessCtx mongo.SessionContext) (interface{}, error) {
		err = idea_dao.IdeaStoreDao.UpdateNameAndWeight(sessCtx, req.IdeaStoreId, req.Name, req.Weight)
		if err != nil {
			return nil, err
		}
		return &idea_store_api.AdminUpdateStoreResponse{
			Code: idea_store_api.AdminUpdateStoreResponse_CODE_OK,
		}, nil
	})
	if err != nil {
		return nil, err
	}
	return ret.(*idea_store_api.AdminUpdateStoreResponse), nil
}

func (apiImpl *IdeaStoreAdminApiImpl) MoveStore(ctx context.Context, req *idea_store_api.AdminMoveStoreRequest) (*idea_store_api.AdminMoveStoreResponse, error) {
	sess, ok := dao.CacheDao.GetAdminSession(ctx, req.AdminSessionId)
	if !ok {
		return &idea_store_api.AdminMoveStoreResponse{
			Code:   idea_store_api.AdminMoveStoreResponse_CODE_WRONG_SESSION,
			ErrMsg: "会话不存在",
		}, nil
	}
	if !sess.AuthOk {
		return &idea_store_api.AdminMoveStoreResponse{
			Code:   idea_store_api.AdminMoveStoreResponse_CODE_NOT_AUTH,
			ErrMsg: "未授权会话",
		}, nil
	}
	adminUserCfg, err := admin_common.GetAdminUserConfig(ctx, apiImpl.adminCfg, sess.UserName)
	if err != nil {
		return nil, err
	}
	if !adminUserCfg.PermInfo.IdeaStorePerm.MoveStore {
		return &idea_store_api.AdminMoveStoreResponse{
			Code:   idea_store_api.AdminMoveStoreResponse_CODE_NO_PERMISSION,
			ErrMsg: "没有权限",
		}, nil
	}

	storeItem, err := idea_dao.IdeaStoreDao.Get(ctx, req.IdeaStoreId)
	if err != nil {
		return &idea_store_api.AdminMoveStoreResponse{
			Code:   idea_store_api.AdminMoveStoreResponse_CODE_NO_STORE,
			ErrMsg: "知识点仓库不存在",
		}, nil
	}
	if storeItem.StoreCateId == req.StoreCateId {
		return &idea_store_api.AdminMoveStoreResponse{
			Code: idea_store_api.AdminMoveStoreResponse_CODE_OK,
		}, nil
	}

	exist, err := idea_dao.IdeaStoreCateDao.Exist(ctx, req.StoreCateId)
	if err != nil {
		return nil, err
	}
	if !exist {
		return &idea_store_api.AdminMoveStoreResponse{
			Code:   idea_store_api.AdminMoveStoreResponse_CODE_NO_STORE_CATE,
			ErrMsg: "知识点仓库分类不存在",
		}, nil
	}

	dbSess, err := dao.StartSession()
	if err != nil {
		return nil, err
	}
	defer dao.EndSession(dbSess)

	ret, err := dbSess.WithTransaction(ctx, func(sessCtx mongo.SessionContext) (interface{}, error) {
		err = idea_dao.IdeaStoreDao.UpdateStoreCate(sessCtx, req.IdeaStoreId, req.StoreCateId)
		if err != nil {
			return nil, err
		}
		{
			count, err := idea_dao.IdeaStoreDao.CountByCate(sessCtx, storeItem.StoreCateId)
			if err != nil {
				return nil, err
			}
			err = idea_dao.IdeaStoreCateDao.UpdateStoreCount(sessCtx, storeItem.StoreCateId, count)
			if err != nil {
				return nil, err
			}
		}
		{
			count, err := idea_dao.IdeaStoreDao.CountByCate(sessCtx, req.StoreCateId)
			if err != nil {
				return nil, err
			}
			err = idea_dao.IdeaStoreCateDao.UpdateStoreCount(sessCtx, req.StoreCateId, count)
			if err != nil {
				return nil, err
			}
		}
		return &idea_store_api.AdminMoveStoreResponse{
			Code: idea_store_api.AdminMoveStoreResponse_CODE_OK,
		}, nil
	})
	if err != nil {
		return nil, err
	}
	return ret.(*idea_store_api.AdminMoveStoreResponse), nil
}

func (apiImpl *IdeaStoreAdminApiImpl) RemoveStore(ctx context.Context, req *idea_store_api.AdminRemoveStoreRequest) (*idea_store_api.AdminRemoveStoreResponse, error) {
	sess, ok := dao.CacheDao.GetAdminSession(ctx, req.AdminSessionId)
	if !ok {
		return &idea_store_api.AdminRemoveStoreResponse{
			Code:   idea_store_api.AdminRemoveStoreResponse_CODE_WRONG_SESSION,
			ErrMsg: "会话不存在",
		}, nil
	}
	if !sess.AuthOk {
		return &idea_store_api.AdminRemoveStoreResponse{
			Code:   idea_store_api.AdminRemoveStoreResponse_CODE_NOT_AUTH,
			ErrMsg: "未授权会话",
		}, nil
	}
	adminUserCfg, err := admin_common.GetAdminUserConfig(ctx, apiImpl.adminCfg, sess.UserName)
	if err != nil {
		return nil, err
	}
	if !adminUserCfg.PermInfo.IdeaStorePerm.RemoveStore {
		return &idea_store_api.AdminRemoveStoreResponse{
			Code:   idea_store_api.AdminRemoveStoreResponse_CODE_NO_PERMISSION,
			ErrMsg: "没有权限",
		}, nil
	}

	storeItem, err := idea_dao.IdeaStoreDao.Get(ctx, req.IdeaStoreId)
	if err != nil { //记录不存在
		return &idea_store_api.AdminRemoveStoreResponse{
			Code: idea_store_api.AdminRemoveStoreResponse_CODE_OK,
		}, nil
	}

	dbSess, err := dao.StartSession()
	if err != nil {
		return nil, err
	}
	defer dao.EndSession(dbSess)

	ret, err := dbSess.WithTransaction(ctx, func(sessCtx mongo.SessionContext) (interface{}, error) {
		err = idea_dao.IdeaStoreDao.Remove(sessCtx, req.IdeaStoreId)
		if err != nil {
			return nil, err
		}
		count, err := idea_dao.IdeaStoreDao.CountByCate(sessCtx, storeItem.StoreCateId)
		if err != nil {
			return nil, err
		}
		err = idea_dao.IdeaStoreCateDao.UpdateStoreCount(sessCtx, storeItem.StoreCateId, count)
		if err != nil {
			return nil, err
		}
		return &idea_store_api.AdminRemoveStoreResponse{
			Code: idea_store_api.AdminRemoveStoreResponse_CODE_OK,
		}, nil
	})
	if err != nil {
		return nil, err
	}
	return ret.(*idea_store_api.AdminRemoveStoreResponse), nil
}
